../../../../models/demos/vit/tests/pcc/test_ttnn_optimized_sharded_vit_wh.py